﻿#region Copyright and Licensing Notice
// Copyright (C) 2011 Steven Bosnick
// 
// This program is open source software; you can redistribute it 
// and/or modifiy it under the terms of the Eclipse Public License,
// version 1.0.
// 
// This program is distributed in the hope that it will be useful, 
// but WITHOUT ANY WARRANTY; without even the implied
// warranty of MERCHANTABILITY or FITNESS FOR A 
// PARTICULAR PURPOSE. See the Eclipse Public License for more
// details.
// 
// You should have received a copy of the Eclipse Public License 
// along with this program; if not, you can find a copy at
// "http://www.opensource.org/licenses/eclipse-1.0.php".
#endregion
using System;
using System.Threading.Tasks;

namespace BitwiseOre.SharedMemory.Utils
{
    internal sealed class TaskEnder<TTask> : ITaskEnder<TTask> where TTask : Task
    {
        public TTask EndTask(IAsyncResult asyncResult, Func<AggregateException, Exception> createException)
        {
            if (asyncResult == null) throw new ArgumentNullException("asyncResult");
            var task = asyncResult as TTask;
            if (task == null) throw new ArgumentException("asyncResult");

            try
            {
                task.Wait();
            }
            catch (AggregateException e)
            {
                throw e.Wrap(createException);
            }
            return task;
        }
    }
}